"""
Case Type   : cm_uninstall
Case Name   : cm_uninstall执行前，集群start状态正常
Create At   : 2025/4/25
Owner       : ningyali
Description :
    1.启动om集群查看状态
    2.取消配置xml文件中的cm相关参数
    3.用户下执行cm_uninstall
    4.cm_ctl查看集群状态
    5.gs_om查看集群状态
    6.查看数据库用户下进程和定时任务
    7.主备节点下查看CMlog目录
    8.主备节点下查看CM数据目录
    9.主备节点下查看CM相关二进制文件
Expect      :
    1.集群start状态正常
    2.执行成功
    3.卸载成功
    4.失败,未安装cm
    5.未安装cm
    6.无cm_server,om_monitor,cm_agent,定时任务
    7.已删除
    8.已删除
    9.已删除
History     :
"""
import os
import unittest

from testcase.utils.ComInstall import CmInstall
from testcase.utils.ComInstall import CmUnInstall
from testcase.utils.Common import Common
from testcase.utils.CommonSH import CommonSH
from testcase.utils.Logger import Logger
from yat.test import Node
from yat.test import macro


class Tools(unittest.TestCase):
    def setUp(self):
        self.log = Logger()
        self.log.info(f'----{os.path.basename(__file__)} start----')
        self.user_node_names = (
            'PrimaryDbUser', 'Standby1DbUser', 'Standby2DbUser')
        self.sh = [CommonSH(n) for n in self.user_node_names]
        self.root_nodes = (Node('PrimaryRoot'), Node('Standby1Root'),
                           Node('Standby2Root'))
        self.user_nodes = [Node(n) for n in self.user_node_names]
        self.common = Common()
        self.install_pkg_path = os.path.dirname(macro.DB_SCRIPT_PATH)
        self.cm = CmInstall(self.install_pkg_path, self.sh[0].node.ssh_user)
        self.un_cm = CmUnInstall(self.install_pkg_path,
                                 self.sh[0].node.ssh_user)

        text = '----prestep1:安装cm;expect:执行成功----'
        self.log.info(text)
        res = self.cm.update_xml_for_cm(*self.root_nodes)
        self.assertGreater(res.count('cmDir'), 1, f'执行失败: {text}')

        res = self.cm.cm_install(self.user_nodes[0])
        self.assertIn('Install CM tool success', res, f'执行失败: {text}')

    def test_cm_uninstall(self):
        text = '----step1:启动om集群查看状态;expect:集群start状态正常----'
        self.log.info(text)
        status = self.sh[0].get_db_cluster_status('detail')
        self.log.info(status)
        if 'stopped' in status:
            start_res = self.sh[0].start_db_cluster()
            self.assertTrue(start_res, f'执行失败: {text}')

        text = '----step2:取消配置xml文件中的cm相关参数 expect:执行成功----'
        self.log.info(text)
        sed_cmd = f'sed -i \'/cm/d\' {self.cm.xml_path};'
        sed_res = self.common.get_sh_result(self.user_nodes[0], sed_cmd)
        self.assertNotIn('error', sed_res.lower(), "执行失败:" + text)

        text = '----step3:用户下执行cm_uninstall expect:卸载成功----'
        self.log.info(text)
        res = self.un_cm.cm_uninstall(
            self.user_nodes[0],
            f'-e {macro.DB_ENV_PATH} --deleteBinary --deleteData')
        self.assertIn('Uninstall CM tool success', res, f'执行失败: {text}')

        text = '----step4:cm_ctl查看集群状态;expect:失败,未安装cm----'
        self.log.info(text)
        show_status = f'source {macro.DB_ENV_PATH}; cm_ctl query -Civdp'
        status_msg = self.common.get_sh_result(self.user_nodes[0], show_status)
        self.assertIn("cm_ctl: command not found", status_msg,
                      "执行失败:" + text)

        text = '----step5:gs_om查看集群状态;expect:未安装cm----'
        self.log.info(text)
        status_msg = self.sh[0].get_db_cluster_status('detail')
        self.log.info(status_msg)
        self.assertNotIn("CMServer State", status_msg, "执行失败:" + text)

        text = '----step6:查看数据库用户下进程和定时任务;' \
               'expect:无cm_server,om_monitor,cm_agent,定时任务----'
        self.log.info(text)
        show_cmd = "crontab -l;ps ux|egrep 'cm_server|cm_agent|om_monitor';"
        process_res = self.sh[0].exec_gs_ssh(show_cmd)
        self.assertNotIn('bin/cm_server', process_res, "执行失败:" + text)
        self.assertNotIn('bin/cm_agent', process_res, "执行失败:" + text)
        self.assertNotIn('om_monitor -L', process_res, "执行失败:" + text)

        text = '----step7:主备节点下查看CMlog目录;expect:已删除----'
        self.log.info(text)
        ls_cmd = f"source {macro.DB_ENV_PATH};ls $GAUSSLOG/cm;"
        ls_res = self.sh[0].exec_gs_ssh(ls_cmd)
        self.assertGreater(ls_res.count('No such file or directory'), 1,
                           "执行失败:" + text)

        text = '----step8:主备节点下查看CM数据目录;expect:已删除----'
        self.log.info(text)
        ls_cmd = f"source {macro.DB_ENV_PATH};ls {self.cm.cm_path};"
        ls_res = self.sh[0].exec_gs_ssh(ls_cmd)
        self.assertGreater(ls_res.count('No such file or directory'), 1,
                           "执行失败:" + text)

        text = '----step9:主备节点下查看CM相关二进制文件;expect:已删除----'
        self.log.info(text)
        ls_cmd = f"source {macro.DB_ENV_PATH};" \
                 f"ls $GAUSSHOME/bin/cm_ctl;" \
                 f"ls $GAUSSHOME/bin/cm_server;" \
                 f"ls $GAUSSHOME/bin/cm_agent;" \
                 f"ls $GAUSSHOME/bin/om_monitor"
        ls_res = self.sh[0].exec_gs_ssh(ls_cmd)
        self.assertGreater(ls_res.count('No such file or directory'), 1,
                           "执行失败:" + text)

    def tearDown(self):
        text = '----poststep1:gs_om查看集群状态;expect:执行成功----'
        self.log.info(text)
        status_msg = self.sh[0].get_db_cluster_status('detail')
        self.log.info(status_msg)
        if 'CMServer State' in status_msg:
            text = '----用例异常cm未卸载再次执行cm_uninstall----'
            self.log.info(text)
            res = self.un_cm.cm_uninstall(
                self.user_nodes[0],
                f'-e {macro.DB_ENV_PATH} --deleteBinary --deleteData')

            status_msg = self.sh[0].get_db_cluster_status('detail')
            self.log.info(status_msg)
            self.assertIn('Uninstall CM tool success', res, f'执行失败: {text}')
        self.assertNotIn("CMServer State", status_msg, "执行失败:" + text)
        self.log.info(f'----{os.path.basename(__file__)} end----')
